package exerise;

import java.util.ArrayList;
import java.util.Arrays;
import java.util.List;

/**
 * 给定一个包含 n 个整数的数组 nums，判断 nums 中是否存在三个元素 a，b，c ，使得 a + b + c = 0 ？找出所有满足条件且不重复的三元组。
 */
public class Ex15 {

    public static void main(String[] args) {
        Ex15 ex15 = new Ex15();
        int[] nums = {-4, -1, -1, -1, 0, 0, 1, 2, 2};
        List<List<Integer>> result = ex15.threeSum(nums);
        System.out.println("result = " + result);
    }

    public List<List<Integer>> threeSum(int[] nums) {
        int sum = 0;
        List<List<Integer>> list = new ArrayList<>();
        int len = nums.length;
        if (len < 3) return list;
        Arrays.sort(nums);
        if (nums[0] > sum) return list;
        for (int k = 0; k < len; k++) {
            if (k > 0 && nums[k] == nums[k - 1]) continue;
            int target = sum - nums[k];
            int i = k + 1;
            int j = len - 1;
            while (i < j) {
                if (nums[i] + nums[j] < target) {
                    i++;
                } else if (nums[i] + nums[j] > target) {
                    j--;
                } else {
                    List<Integer> data = Arrays.asList(nums[k], nums[i], nums[j]);
                    list.add(data);
                    while (i + 1 < j && nums[i] == nums[i + 1]) i++;
                    while (j - 1 > i && nums[j] == nums[j - 1]) j--;
                    i++;
                    j--;
                }
            }
        }
        return list;
    }
}
